from typing import Optional, Union
from osc_transformers import TransformerDecoder, Tokenizer
from pathlib import Path


class LLM:
    def __init__(
        self,
        chat_prompt: Optional[str] = None,
        checkpoint_dir: Optional[Union[str, Path]] = None,
        
    ):
        self.tokenizer = Tokenizer(checkpoint_dir=checkpoint_dir)